*! version 6.3 5NOV2019 DIME Analytics dimeanalytics@worldbank.org

cap program drop 	ieddtab
	program define	ieddtab, rclass

	syntax varlist(numeric) [if] [in] [aw fw iw pw], ///
					///
		Time(varname numeric) TREATment(varname numeric) 	///
		[ 													///
		/* Regression options */							///
		COVARiates(varlist numeric ts fv)					///
		vce(string) 										///
															///
		/* Output display */								///
		STARLevels(numlist descending min=3 max=3 >0 <1)	///
		stardrop											///
		ROWLabtype(string) 									///
		rowlabtext(string)									///
		ERRortype(string)									///
		format(string)										///
		onerow												///
		ADDNotes(string)									///
		nonotes												///
															///
		/* Tex output options */							///
		SAVETex(string)										///
		replace												///
		TEXDOCument											///
		TEXCaption(string)									///
		TEXLabel(string)									///
		TEXNotewidth(numlist min=1 max=1)					///
		texvspace(string)									///
		nonumbers											///
		]

		*Set minimum version for this command
		version 11

	/*************

		Input handling

	*************/

	preserve

	*Remove observations excluded by if and in
	marksample touse,  novarlist
	keep if `touse'

	*TIME AND TREATMENT NOT IN OUTCOMEVARS

	*Test that the variables listed in time() and treatment() is not also in the main varlist
	if `:list time in varlist' != 0 {
		noi di as error "{phang}The variable `time' listed in option {inp:time(`time')} is also listed in the outcome variables which is not allowed.{p_end}"
		error 198
	}
	if `:list treatment in varlist' != 0 {
		noi di as error "{phang}The variable `treatment' listed in option {inp:treatment(`treatment')} is also listed in the outcome variables which is not allowed.{p_end}"
		error 198
	}

	*LABELS
	*Test and prepare the row lables and test how long the longest label is.
	prepRowLabels `varlist', rowlabtype("`rowlabtype'") rowlabtext("`rowlabtext'")
	local rowlabels "`r(rowlabels)'"
	local labmaxlen "`r(rowlabel_maxlen)'"

	*ERRORTYPES
	*If error type is used test that it is only one word and that word is an allowed error type
	if "`errortype'" != "" {
		local errortype = lower("`errortype'")
		if (`: word count `errortype'' != 1) | !inlist("`errortype'", "sd", "se", "errhide") {
			noi di as error "{phang}Value in option errortype(`errortype') can only one out of the following words; se, sd, errhide.{p_end}"
			error 198
		}
	}
	else {
		*No errortype specified use default which is standard errors
		local errortype "se"
	}

	*VCE OPTIONS
	if "`vce'" != "" {
		local vce_nocomma = subinstr("`vce'", "," , " ", 1)

		tokenize "`vce_nocomma'"
		local vce_type `1'

		if "`vce_type'" == "robust" {

			*Robust is allowed and not other tests needed
		}
		else if "`vce_type'" == "cluster" {

			*Create a local for displaying number of clusters
			local cluster_var `2'

			cap confirm variable `cluster_var'

			if _rc {

				*Error for vce(cluster) incorrectly applied
				noi display as error "{phang}The cluster variable in vce(`vce') does not exist or is invalid for some other reason. See {help vce_option :help vce_option} for more information. "
				error _rc

			}

			*Add a local names cluster to be passed to commands
			local cluster cluster
		}
		else if  "`vce_type'" == "bootstrap" {

			*bootstrap is allowed and not other tests needed. Error checking is more comlex, add tests here in the future.
		}
		else {

			*Error for vce() incorrectly applied
			noi display as error "{phang}The vce type `vce_type' in vce(`vce') is not allowed. Only robust, cluster or bootstrap are allowed. See {help vce_option :help vce_option} for more information."
			error 198

		}

			local error_estm 		vce(`vce')

		if "`vce_type'" != "robust" {
			local error_estm_mean 	vce(`vce')
		}
	}

	*DEFAULT STAR LEVELS
	*Default star levels if option not used
	if "`starlevels'" == "" local starlevels ".1 .05 .01"

	** If the format option is specified, then test if there is a valid format specified
	if "`format'" != "" {

			** Creating a numeric mock variable that we attempt to apply the format
			*  to. This allows us to piggy back on Stata's internal testing to be
			*  sure that the format specified is at least one of the valid numeric
			*  formats in Stata
				tempvar  formattest
				gen 	`formattest' = 1
			cap	format  `formattest' `format'

			** Some error with the format
			if _rc == 120 {
				di as error "{phang}The format specified in format(`format') is not a valid Stata format. See {help format} for a list of valid Stata formats. This command only accepts the f, fc, g, gc and e format.{p_end}"
				error 120
			}
			else if _rc != 0 {
				di as error "{phang}Something unexpected happened related to the option format(`format'). Make sure that the format you specified is a valid format. See {help format} for a list of valid Stata formats. If this problem remains, please report this error at https://github.com/worldbank/ietoolkit.{p_end}"
				error _rc
			}

			** Format is a valid format, but is it one we allow
			else {

				local fomrmatAllowed 0
				local charLast  = substr("`format'", -1,.)
				local char2Last = substr("`format'", -2,.)

				if  "`charLast'" == "f" | "`charLast'" == "e" {
					local fomrmatAllowed 1
				}
				else if "`charLast'" == "g" {
					if "`char2Last'" == "tg" {
						*format tg not allowed. all other valid formats ending on g are allowed
						local fomrmatAllowed 0
					}
					else {

						*Formats that end in g that is not tg can only be g which is allowed.
						local fomrmatAllowed 1
					}
				}
				else if  "`charLast'" == "c" {
					if "`char2Last'" != "gc" & "`char2Last'" != "fc" {
						*format ends on c but is neither fc nor gc
						local fomrmatAllowed 0
					}
					else {

						*Formats that end in c that are either fc or gc are allowed.
						local fomrmatAllowed 1
					}
				}
				else {
					*format is neither f, fc, g, gc nor e
					local fomrmatAllowed 0
				}
				if `fomrmatAllowed' == 0 {
					di as error "{phang}The format specified in format(`format') is not allowed. Only format f, fc, g, gc and e are allowed. See {help format} for details on Stata formats.{p_end}"
					error 120
				}
			}
		}
		else {
			*Default value if fomramt not specified
			local format = "%9.2f"
		}


	/*************

		Initiate the result matrix

	*************/

	*Creates the template for the result matrix in a subfunction
	templateResultMatrix
	mat startRow = r(startRow)
	local colnames = "`r(colnames)'"

	*Remove this when ready for production
	//noi di "Start row to see headers, remove for production"
	//matlist startRow // See the default row with its column names

	*Initiate the result matrix with a place holder row that will not be used for anything. Matrices cannot be initiated empty
	mat ddtab_resultMap = startRow
	mat rownames ddtab_resultMap = placeholder


	/*************

		Loop over all variables and prepare the data

	*************/

	foreach var of local varlist {

		//noi di "Variable `var'"

		*Each variable is a row in the result matrix
		mat `var' = startRow
		mat rownames `var' = `var'
		mat colnames `var' = `colnames'

		*Local that keeps track of which column to fill
		local colindex 0

		tempvar regsample interactionvar

		/*************

			Calculating 2nd difference and restict the sample
			for this output var

		*************/

		*Generate the interaction var
		gen `interactionvar' = `treatment' * `time'

		*Run the regression to get the double difference
		qui reg `var' `treatment' `time' `interactionvar' `covariates' [`weight'`exp'], `error_estm'
		mat resTable = r(table)

		**This is why this is done first. All other calculations
		* for this outcome var should be restricted to this sample.
		gen `regsample' = e(sample)
		local N `e(N)'

		**Test that the dummy vars creates for valid groups for each
		* combination of time and treat in the sample used for this outcome var
		testDDdums `time' `treatment' `regsample' `var'  //comment in when this is made into a command

		*Get the second differnce
		local ++colindex
		mat `var'[1,`colindex'] =  el(resTable,1,3)

		*Get the standard error of second difference
		local ++colindex
		convertErrs el(resTable,2,3) `N' "`errortype'"
		mat `var'[1,`colindex'] =  `r(converted_error)'

		*Get the number of stars using sub-command countStars
		local ++colindex
		local pvalue = el(resTable,4,3)
		countStars `pvalue' `starlevels' `stardrop'
		mat `var'[1,`colindex'] = `r(stars)'

		*Get the N of second difference regression
		local ++colindex
		mat `var'[1,`colindex'] = `N'

		**Increment column index regardless if clusters is
		* used. If it is not used column should be missing
		local ++colindex
		if "`vce_type'" == "cluster" {
			*Add the number of clusters if clusters are used
			mat `var'[1,`colindex'] =  `e(N_clust)'
		}

		/*************

			Calculating 1st differences

		*************/
		forvalues tmt01 = 0/1 {

			*Regress time against the outcome var one tmt group at the time
			qui reg `var' `time' `covariates' if `treatment' == `tmt01' & `regsample' == 1 [`weight'`exp'], `error_estm'
			mat resTable = r(table)

			//Get the 1st diff
			local ++colindex
			mat `var'[1,`colindex'] =  el(resTable,1,1)

			*Get the standard error of 1st diff
			local ++colindex
			convertErrs el(resTable,2,1) `e(N)' "`errortype'"
			mat `var'[1,`colindex'] =  `r(converted_error)'

			*Get the number of stars using sub-command countStars
			local ++colindex
			local pvalue = el(resTable,4,1)
			countStars `pvalue' `starlevels' `stardrop'
			mat `var'[1,`colindex'] = `r(stars)'

			*Get the N of first difference regression
			local ++colindex
			mat `var'[1,`colindex'] =  `e(N)'

			**Increment column index regardless if clusters is
			* used. If it is not used column should be missing
			local ++colindex
			if "`vce_type'" == "cluster" {
				*Add the number of clusters if clusters are used
				mat `var'[1,`colindex'] =  `e(N_clust)'
			}
		}


		/*************

			Calculating standard means for all groups

		*************/
		forvalues tmt01 = 0/1 {
			forvalues t01 = 0/1 {

				*Summary stats on this group
				qui mean `var' if `treatment' == `t01' & `time' == `tmt01' & `regsample' == 1 [`weight'`exp'], `error_estm_mean'

				mat resTable = r(table)

				//Get the mean
				local ++colindex
				mat `var'[1,`colindex'] =  el(resTable,1,1)

				*Get the standard error of the mean
				local ++colindex
				convertErrs el(resTable,2,1)  `e(N)' "`errortype'"
				mat `var'[1,`colindex'] =  `r(converted_error)'

				*Get the N of the ordinary means regressions
				local ++colindex
				mat `var'[1,`colindex'] =  `e(N)'

				**Increment column index regardless if clusters is
				* used. If it is not used column should be missing
				local ++colindex
				if "`vce_type'" == "cluster" {
					*Add the number of clusters if clusters are used
					qui tab `cluster_var' if `treatment' == `t01' & `time' == `tmt01' & `regsample' == 1
					mat `var'[1,`colindex'] =  `r(r)'
				}
			}
		}

		*Append this row to the result table
		mat 	ddtab_resultMap = (ddtab_resultMap \ `var')

	}

	*Remove placeholder row
	matrix ddtab_resultMap = ddtab_resultMap[2..., 1...]

	*Returning the result matrix for advanced users to do their own thing with
	mat returnMat = ddtab_resultMap
	return matrix ieddtabResults returnMat

	*Show the final matrix will all data needed to start building the output
	//noi di "Matlist with results"
	//matlist ddtab_resultMap

	/*************

		Test the matrix

	*************/

	*Test that onerow option is valid, i.e. the N in each column is the same across all rows

	if "`onerow'" != "" {
		noi testonerow `varlist', ddtab_resultMap(ddtab_resultMap) `cluster'
	}

	/***************************************

		Prepare notes

	***************************************/

	*Prepare the automated table not unless it was disabled by option nonote
	if "`notes'" == "" {

		local note_obs	"The baseline means only include observations not omitted in the 1st and 2nd differences. The number of observations in the 1st and 2nd differences includes both baseline and follow-up observations."

		*Only include note on stars levels if stardrop was NOT used
		if "`stardrop'" == "" {
			local star1_value : word 1 of `starlevels'
			local star2_value : word 2 of `starlevels'
			local star3_value : word 3 of `starlevels'
			local note_stars "***, **, and * indicate significance at the `star3_value', `star2_value', and `star1_value' percent critical level."
		}

		*Only include note on covariates if covariates were used
		if "`covariates'" != "" {
			local note_cov	"The following variable(s) was included as covariates [`covariates']."
		}

		if "`vce'" != "" & "`errortype'" != "errhide" {

			*Display variation in Standard errors (default) or in Standard Deviations
			if "`errortype'" == "se" {
				*Standard Errors string
				local 	variance_type_name 	"standard errors"
			}
			else if "`errortype'" == "sd" {
				*Standard Deviation string
				local 	variance_type_name 	"standard deviations"
			}

			if "`vce_type'" == "robust"		local note_error	"1st and 2nd difference `variance_type_name' are robust. "
			if "`vce_type'" == "cluster"  	local note_error	"All columns display `variance_type_name' clustered at variable `cluster_var'. "
			if "`vce_type'" == "bootstrap"  local note_error	"All columns display `variance_type_name' estimated using bootstrap. "
		}

		* Add note on weights
		if "`weight'" != "" {
			local weightvar = subinstr("`exp'", "=", "", .)
			local weightvar = stritrim(strtrim(`"`weightvar'"'))

			noi di "`weight'"
				 if "`weight'" == "aweight" local weightopt analytical
			else if "`weight'" == "fweight" local weightopt frequency
			else if "`weight'" == "pweight" local weightopt probability
			else if "`weight'" == "iweight" local weightopt importance

			local note_weight "Variable `weightvar' used as `weightopt' weight. "
		}

		local note `"`note_obs' `note_stars' `note_cov' `note_error' `note_weight'"'

	}

	**if a manual note was added in addnotes(), combine the manually added
	* addnotes to note (which is still empty if nonotes was used)
	if "`addnotes'" != "" {
		local note	`"`addnotes' `note'"'
	}

	*Remove leading, trailing or multiple spaces from the string.
	local note = stritrim(strtrim(`"`note'"'))


	/*************

		Output table in result window

	*************/

	outputwindow `varlist' , ddtab_resultMap(ddtab_resultMap) labmaxlen(`labmaxlen') rwlbls(`rowlabels') ///
		 `errortype' format(`format') note(`note') `cluster'

	/*************

		Output table in LaTeX

	*************/
	if "`savetex'" != "" {

		outputtex `varlist', 	ddtab_resultMap(ddtab_resultMap) 	///
								savetex(`savetex') `replace'  ///
								`texdocument' texcaption("`texcaption'") texlabel("`texlabel'") texnotewidth(`texnotewidth') ///
								`onerow' format(`format') rwlbls("`rowlabels'") errortype(`errortype') ///
								note(`note') texvspace("`texvspace'") `cluster' `numbers'

	}

	restore

end


/***************************************
****************************************

	Write sub-commands for input handling

****************************************
***************************************/

**Program to prepare row labels for each outcome var, using variable name,
* variable label or manually entered labels depedning on user input. Variable
* names are used if no input.
cap program drop 	prepRowLabels
	program define	prepRowLabels, rclass

	syntax varlist, [rowlabtype(string) rowlabtext(string)]

	*First test input
	if "`rowlabtype'" == "varlab" | "`rowlabtype'" == "" | "`rowlabtype'" == "varname" {
		//All is good do nothing
	}
	else {
		noi display as error "{phang}Row label type [`rowlabtype'] is not a valid row label type. Enter either {it:varlab} or {it:varname} (the default if not specified is {it:varname}). See option {help ieddtab:rowlabtype} for details."
		error 198
	}

	/*************
		Parse through the manually entered labels
	*************/

	*Create a local with the rowlabel input to be tokenized
	local row_labels_to_tokenize `rowlabtext'

	while "`row_labels_to_tokenize'" != "" {

		*Parsing name and label pair
		gettoken nameAndLabel row_labels_to_tokenize : row_labels_to_tokenize, parse("@@")

		*Splitting name and label
		gettoken name label : nameAndLabel

		/*
			We store two locals, one with varnames and one with label. They are
			stored in the same order, so if we find varname we know we will find
			the corresponding label in the same order in the other local.
		*/

		*** Tests

		*Checking that the variables used in rowlabels() are included in the table
		local name_correct : list name in varlist
		if `name_correct' == 0 {
			noi display as error "{phang}Variable [`name'] listed in rowlabtext(`rowlabtext') is not found among the outcome variables."
			error 111
		}

		*Testing that no label is missing
		if "`label'" == "" {
			noi display as error "{phang}For variable [`name'] listed in rowlabtext(`rowlabels') you have not specified any label. Labels are required for all variables listed in rowlabels(). Variables omitted from rowlabtext() will be assigned labels according to the rule in rowlabtype(). See also option {help ieddtab:rowlabtext}"
			error 198
		}

		*Storing the name in local to be used to get index of corresponding label in the second local
		local rowLabelNames `"`rowLabelNames' "`name'" "'

		*Removing leading or trailing spaces and store label with the same index as in the name local above
		local label = trim("`label'")
		local rowLabelLabels `"`rowLabelLabels' "`label'" "'


		*Parse char is not removed by gettoken, so remove it and start over
		local row_labels_to_tokenize = subinstr("`row_labels_to_tokenize'" ,"@@","",1)
	}

	**Set up locals to be returned. One with row labels without varname but in
	* final order, and one local with length of longest label
	local preppedRowLabels
	local maxLen 0

	*Loop over all varaibles and prepare the labels
	foreach var of local varlist {

		**Get the index of this variable in manually entered labels. Index is
		* zero if no manually entered label for this variable
		local rowLabPos : list posof "`var'" in rowLabelNames

		if `rowLabPos' == 0 {

			*No label was manually entered for this variable, use the other rules
			if "`rowlabtype'" == "varlab" {
				local this_label : variable label `var'		//Use variable label
			}
			else {

				local this_label `var'						//Default, use varname

			}
		}
		else {
			*Getting the manually defined label corresponding to this variable
			local this_label : word `rowLabPos' of `rowLabelLabels'
		}

		*Update the two locals that will be returned
		local preppedRowLabels "`preppedRowLabels' @@`this_label'"	//Prepare the final list
		local maxLen = max(strlen("`this_label'"),`maxLen')	//Store the length on the longest label, used on some outputs
	}

	*Return the locals
	return local rowlabels `preppedRowLabels'
	return local rowlabel_maxlen `maxLen'

end

** Program that test the t and tmt variables are valid dummies
*  for a Diff-in-Diff. I.e. they are both dummies and there are
*  at least 2 observations in each combination (0,0), (1,0),
*  (0,1) and (1,1)
cap program drop 	testDDdums
	program define	testDDdums

		args time treat samplevar outputvar

		*Test taht the dummies are dummies
		foreach dummy in `time' `treat' {

			cap assert inlist(`dummy',1,0) | missing(`dummy') | `samplevar' == 0
			if _rc {
				noi di as error "{phang}In the differences-in-differences regression for outputvar [`outputvar'], the dummy [`dummy'] must contain only values 1 and 0. See tab below for a list of other values detected. The tab is restricted to the observations that are not excluded due to missing values etc. in the differences-in-differences regression.{p_end}""
				noi tab `dummy' if `samplevar' == 1, missing
				error 480
			}
		}

		**Test that for only two of three dummies there are observations
		* that has only that dummy. I.e. the two that is not the
		* interaction. If the interaction is 1, all three shluld be 1.

		*Test that there is some observations in each group
		forvalues tmt01 = 0/1 {
			forvalues t01 = 0/1 {

				*Summary stats on this group
				qui count if `treat' == `t01' & `time' == `tmt01' & `samplevar' == 1

				//There got to be more than 1 observation in each group for the regression to make sense
				if `r(N)' < 2 {
					noi di as error "{phang}In the differences-in-differences regression for outputvar [`outputvar'] there were not enough observations in each combination of treatment [`treat'] and time [`time']. See tab below for groups with insufficient observations. The tab below is restricted to the observations that are not excluded due to missing values etc. in the differences-in-differences regression.{p_end}""
					noi tab `treat' `time' if `samplevar' == 1, missing
					error 480

				}
			}
		}

	end



/***************************************
****************************************

	Write sub-commands for stats

****************************************
***************************************/

*Sets up template for the result matrix
cap program drop 	templateResultMatrix
	program define	templateResultMatrix, rclass


	*Set the names of the different columns in the result Matrix
	local 2ndDiff_cols 			2D 2D_err 2D_stars 2D_N 2D_clus

	local 1stDiff_C_cols		1DC 1DC_err 1DC_stars 1DC_N 1DC_clus
	local 1stDiff_T_cols		1DT 1DT_err 1DT_stars 1DT_N 1DT_clus

	local basicMean_C0_cols		C0_mean C0_err C0_N C0_clus
	local basicMean_T0_cols		T0_mean T0_err T0_N T0_clus
	local basicMean_C1_cols		C1_mean C1_err C1_N C1_clus
	local basicMean_T1_cols		T1_mean T1_err T1_N T1_clus

	local colnames `2ndDiff_cols' `1stDiff_C_cols' `1stDiff_T_cols' `basicMean_C0_cols' `basicMean_T0_cols' `basicMean_C1_cols' `basicMean_T0_cols'

	*Define default row here. The results for each var will be one row that starts with all missing vlaues
	mat startRow = (.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.,.)
	mat colnames startRow = `colnames'

	return matrix startRow startRow
	return local colnames `colnames'

end

*Take the significance levels and count number of starts
cap program drop 	countStars
	program define	countStars, rclass

	args pvalue star1 star2 star3 stardrop

	local stars 0

	*Option to suppress all stars
	if "`stardrop'" == "" {
		foreach star_p_level in `star1' `star2' `star3' {

			if `pvalue' < `star_p_level' local ++stars
		}
	}

	return local stars `stars'

end

**Convert the errors to the display stats picked
* by the user. Standard errors is the default.
cap program drop 	convertErrs
	program define	convertErrs, rclass

	args se_err N errortype

	*Test if custom star levels are used
	if "`errortype'" == "sd" {
		local err = `se_err' * sqrt(`N')
	}
	else {
		//For se keep se, for errhide keep se as matrix must have some value even if it will not be used
		local err = `se_err'
	}

	return local converted_error `err'

end

cap program drop 	testonerow
	program define	testonerow, rclass

	syntax varlist, ddtab_resultMap(name) [cluster]

qui {
	local numVars :word count `varlist'

	*List of all columns that must be the same in case onerow is used
	local ncols 2D_N 1DC_N 1DT_N C0_N T0_N

	*Add cluster
	if "`cluster'" == "cluster" {
		local ncols `ncols' 2D_clus 1DC_clus 1DT_clus C0_clus T0_clus
	}

	*Loop over the columns with N
	foreach ncolname of local ncols {

		*Loop over the rows in the matrix
		forvalues row = 1/`numVars' {

			*Get the value from the matrix
			mat temp = ddtab_resultMap[`row', "`ncolname'"]
			local ncol = el(temp,1,1)

			*If it is the first row, save to be compared to later rows
			if `row' == 1 {
				local n = `ncol'
			}

			*If not the first row, compare this row to the first row, if it is not the same for all rows, then the option onerow is not valid.
			else if `n' != `ncol' {

				*Split the n colname to column (2D, C0, 1DT etc.) and stat (N or clus)
				tokenize "`ncolname'", parse("_")
				local onerowcol `1'
				local onerowstat `3'

				*Prepare string with explanatory column name
				if "`onerowcol'" == "2D"	local colstring "2nd difference regression"
				if "`onerowcol'" == "1DC"	local colstring "1st difference regression in control group"
				if "`onerowcol'" == "1DT"	local colstring "1st difference regression in treatment group"
				if "`onerowcol'" == "C0"	local colstring "mean of control group in time = 0"
				if "`onerowcol'" == "T0"	local colstring "mean of treatment group in time = 0"
				if "`onerowcol'" == "C1"	local colstring "mean of control group in time = 1"
				if "`onerowcol'" == "T1"	local colstring "mean of treatment group in time = 1"

				*Prepare string with stat name
				if "`onerowstat'" == "N"	local statstring "observations"
				if "`onerowstat'" == "clus" local statstring "clusters"

				*Name of the variables
				local firstVar : word 1 	of `varlist'
				local thisVar  : word `row' of `varlist'

				noi di as error "{phang}There are different number of `statstring' in the variables `firstVar' and `thisVar' in the `colstring'. The number of `statstring' for each statistic must be same in all variables for the option {inp:onerow} to be valid. Either remove the {inp:onerow} option or investigate why the number is different accross variables.{p_end}"
				error 480
			}
		}
	}
}
end


/***************************************
****************************************

	Write sub-commands for outputs

****************************************
***************************************/


	/******
	The result matrix can be passed into subcommands that output in either Excel, LaTeX or in the main window.

	The name of the result matrix is ddtab_resultMap. It can be referenced like this ddtab_resultMap[row, col] where
	row is the variable order where 1 is the first varaible in the varlist and col is the name of the stat.

	You always must make it first in to a 1x1 matrix, and then make that a local.

	*/

	*The results can be accessed like this, which makes the sub-commands less sensitive to changing column order in the section above.
	//mat A = ddtab_resultMap[3, "C0_Mean"] // returns a 1x1 matrix with the baseline mean for the countrol grop for the thrid outcome var
	//local a = el(A,1,1)				// returns the value

	//matlist A
	//di `a'

	/*
		Column name dictionary
			Differences:
			- 2D : Second difference coefficient (t*tmt == 1)
			- 1DC : First  difference coefficient control (t == 0)
			- 1DT : First  difference coefficient treatment (tmt == 0)

			for each coefficent these stats are also provided:
				- _err : Second difference errors (type of errors is set in command errortype)
				- _stars :  Second difference - The number of significance stars (sig level set in command)
				- _N : Second difference - Number of observtions in the regression
				- _clus : number of clusters (missing value if cluster wsa not used)

			Group means:
			- C0 - Control time 0
			- T0 - Treatment time 0
			- C1 - Control time 1
			- T1 - Treatment time 1

			for each group these stats are also provided:
				- _mean : the mean of the group
				- _err : the error in the mean (type of errors is set in command errortype)
				- _N : number of observations in baseline means
				- _clus : number of clusters (missing value if cluster wsa not used)

	*/


	/*************

		Output table in result window

	*************/

	cap program drop 	outputwindow
		program define	outputwindow

		syntax varlist , ddtab_resultMap(name) labmaxlen(numlist) rwlbls(string)  format(string) ///
			[errhide sd se note(string) cluster]

		*Prepare lables for the erorrs to be displayed (in case any)
		if "`sd'" != "" local errlabel "SD"
		if "`se'" != "" local errlabel "SE"

		*Count numbers of variables to loop over
		local numVars = `:word count `varlist''

		*List of variabls to display and loop over when formatting
		local statlist 2D 1DT 1DC C0_mean T0_mean 2D_err 1DT_err 1DC_err C0_err T0_err 2D_N 1DT_N 1DC_N C0_N T0_N

		*Add cluster columns to the list if clusters were used
		if "`cluster'" == "cluster" {
			local statlist `statlist'  2D_clus 1DT_clus 1DC_clus C0_clus T0_clus
		}

		*************************
		* Table width for label column

		** Add minumum lenght in case all row titles are shorter
		*  than "Variable" that is the title of this column
		local labmaxlen = max(`labmaxlen', 8)

		local first_hhline = 2 + `labmaxlen'
		local first_col = 4 + `labmaxlen'

		*************************
		* Table width for baseline columns

		local bsln_space = 2
		local bsln_stat_left = `bsln_space' + 1
		local bsln_width = ((`bsln_space' * 2) + 8)

		*************************
		* Table width for first difference column

		local diff_space = 3
		local diff_stat_left = `diff_space' + 1
		local diff_width = ((`diff_space' * 2) + 10)

		*************************
		* Table width control column

		local ctrl_hline = `bsln_width' + 1 + `diff_width'
		local ctrl_space = (`ctrl_hline' - 7) / 2

		*************************
		* Table width treatment column

		local tmt_hline = `bsln_width' + 1 + `diff_width'
		local tmt_space = (`tmt_hline' - 9) / 2

		*************************
		* Table width diff-in-diff column

		local didi_stat_left = 4
		local diffdiff_width  = 17

		*************************
		* Calculate stats column indices

		local bsln_c_col	= `first_col'  + `bsln_width' + 1
		local diff_c_col	= `bsln_c_col' + `diff_width' + 1
		local bsln_t_col	= `diff_c_col' + `bsln_width' + 1
		local diff_t_col	= `bsln_t_col' + `diff_width' + 1
		local didi_col		= `diff_t_col' + `diffdiff_width' + 1


		*************************
		* Start writing table

		*Three title rows
		noi di as text "{c TLC}{hline `first_hhline'}{c TT}{hline `ctrl_hline'}{c TT}{hline `tmt_hline'}{c TT}{hline 17}{c TRC}"
		noi di as text "{c |}{col `first_col'}{c |}{dup `ctrl_space': }Control{dup `ctrl_space': }{c |}{dup `tmt_space': }Treatment{dup `tmt_space': }{c |}  Difference-in  {c |}"
		noi di as text "{c |}{col `first_col'}{c |}{dup `bsln_space': }Baseline{dup `bsln_space': }{c |}{dup `diff_space': }Difference{dup `diff_space': }{c |}{dup `bsln_space': }Baseline{dup `bsln_space': }{c |}{dup `diff_space': }Difference{dup `diff_space': }{c |}   -difference   {c |}"

		*Stats titles, show the stats displayed for each column in the order they are displayed
		noi di as text "{c |}{col `first_col'}{c |}{dup `bsln_stat_left': } Mean{col `bsln_c_col'}{c |}{dup `diff_stat_left': } Coef.{col `diff_c_col'}{c |}{dup `bsln_stat_left': } Mean{col `bsln_t_col'}{c |}{dup `diff_stat_left': } Coef.{col `diff_t_col'}{c |}{dup `didi_stat_left': } Coef.{col `didi_col'}{c |}"

		*Display error type in title unless errhide was used
		if "`errhide'" == "" {
			noi di as text "{c |}{col `first_col'}{c |}{dup `bsln_stat_left': }(`errlabel'){col `bsln_c_col'}{c |}{dup `diff_stat_left': }(`errlabel'){col `diff_c_col'}{c |}{dup `bsln_stat_left': }(`errlabel'){col `bsln_t_col'}{c |}{dup `diff_stat_left': }(`errlabel'){col `diff_t_col'}{c |}{dup `didi_stat_left': }(`errlabel'){col `didi_col'}{c |}"
		}

		*Display cluster in title if clusters were used
		if "`cluster'" == "cluster" {
			noi di as text "{c |}{col `first_col'}{c |}{dup `bsln_stat_left': }Clusters{col `bsln_c_col'}{c |}{dup `diff_stat_left': }Clusters{col `diff_c_col'}{c |}{dup `bsln_stat_left': }Clusters{col `bsln_t_col'}{c |}{dup `diff_stat_left': }Clusters{col `diff_t_col'}{c |}{dup `didi_stat_left': }Clusters{col `didi_col'}{c |}"
		}

		*Last row with N in title as well as "Variable" in the first column
		noi di as text "{c |}{col 3}Variable{col `first_col'}{c |}{dup `bsln_stat_left': } N{col `bsln_c_col'}{c |}{dup `diff_stat_left': } N{col `diff_c_col'}{c |}{dup `bsln_stat_left': } N{col `bsln_t_col'}{c |}{dup `diff_stat_left': } N{col `diff_t_col'}{c |}{dup `didi_stat_left': } N{col `didi_col'}{c |}"

		*Bottom row to table header
		noi di as text "{c LT}{hline `first_hhline'}{c +}{hline `bsln_width'}{c +}{hline `diff_width'}{c +}{hline `bsln_width'}{c +}{hline `diff_width'}{c +}{hline 17}{c RT}"


		*Loop over each variable and prepare the row
		forvalues row = 1/`numVars' {

			*Get labels from the list of labels previously prepared
			local rwlbls = trim(subinstr("`rwlbls'", "@@","", 1))
			gettoken label rwlbls : rwlbls, parse("@@")

			*Loop over all the stats for this variable
			foreach stat of local statlist {

				**Run sub command that gets value from matrix and prepares it
				* in the format suitable for the result window table (and adding
				* stars if applicable)
				windowdiformat , statname("`stat'") row(`row') ddtab_resultMap(ddtab_resultMap) format("`format'") bslnw(`bsln_width') diffw(`diff_width') ddw(`diffdiff_width')

				*The main stat
				local `stat' `r(disp_stata)'

				*The number of spaces before the stat in the colum
				local `stat'_space = `r(disp_pre_space)'
			}

			*Disaplay each variable row at the same time
			noi di as text "{c |} `label'{col `first_col'}{c |}{dup `C0_mean_space': }`C0_mean'{dup `1DC_space': }`1DC'{dup `T0_mean_space': }`T0_mean'{dup `1DT_space': }`1DT'{dup `2D_space': }`2D'"

			*Unless error type is errhide, show the errors on a seperate row
			if "`errhide'" == "" {
				noi di as text "{c |}{col `first_col'}{c |}{dup `C0_err_space': }`C0_err'{dup `1DC_err_space': }`1DC_err'{dup `T0_err_space': }`T0_err'{dup `1DT_err_space': }`1DT_err'{dup `2D_err_space': }`2D_err'"
			}

			*Unless error type is errhide, show the errors on a seperate row
			if "`cluster'" == "cluster" {
				noi di as text "{c |}{col `first_col'}{c |}{dup `C0_clus_space': }`C0_clus'{dup `1DC_clus_space': }`1DC_clus'{dup `T0_clus_space': }`T0_clus'{dup `1DT_clus_space': }`1DT_clus'{dup `2D_clus_space': }`2D_clus'"
			}

			*The number of observations are shown on a separate row
			noi di as text "{c |}{col `first_col'}{c |}{dup `C0_N_space': }`C0_N'{dup `1DC_N_space': }`1DC_N'{dup `T0_N_space': }`T0_N'{dup `1DT_N_space': }`1DT_N'{dup `2D_N_space': }`2D_N'"


		}

		*Add bottom notes to the table
		noi di as text "{c BLC}{hline `first_hhline'}{c BT}{hline `bsln_width'}{c BT}{hline `diff_width'}{c BT}{hline `bsln_width'}{c BT}{hline `diff_width'}{c BT}{hline 17}{c BRC}"

		*************************
		* Write notes below the table
		if (`"`note'"' != "") {
			noi di as text `"{pstd}`note'{p_end}"'
		}

	end


cap program drop 	windowdiformat
	program define	windowdiformat, rclass

	syntax , statname(string) row(numlist) ddtab_resultMap(name) format(string)  bslnw(numlist) diffw(numlist) ddw(numlist)

		local numSpace 0

		mat temp = ddtab_resultMap[`row', "`statname'"]
		local `statname' = el(temp,1,1)

		if substr("`statname'", -2,.) == "_N" | substr("`statname'", -5,.) == "_clus" {
			local `statname'	: display %9.0f ``statname''
		}
		else {
			local `statname' 	: display `format' ``statname''
		}

		*Trim spaces on left from left.
		local `statname' = ltrim("``statname''")

		*For coefficients, add stars if applicable
		if inlist("`statname'", "2D", "1DT", "1DC") {

			mat starNumMat = ddtab_resultMap[`row', "`statname'_stars"]
			local starNum = el(starNumMat,1,1)

			if `starNum' == 0 local `statname' "``statname''    "
			if `starNum' == 1 local `statname' "``statname''*   "
			if `starNum' == 2 local `statname' "``statname''**  "
			if `starNum' == 3 local `statname' "``statname''*** "
		}

		*Add brackets to errors
		if inlist("`statname'", "2D_err", "1DT_err", "1DC_err", "C0_err", "T0_err") {
			local `statname' "(``statname'')"
		}

		*Get the length of the characters to display
		local len = strlen("``statname''")

		*Which column for this stat
		local colName = substr("`statname'", 1, 2)

		*Get corresponding width for that col
		if "`colName'" == "C0" local colw = `bslnw'
		if "`colName'" == "T0" local colw = `bslnw'
		if "`colName'" == "1D" local colw = `diffw'
		if "`colName'" == "2D" local colw = `ddw'

		* Calculate the number of spaces needed before
		* the value and add spaces after it
		if inlist("`statname'", "C0_mean", "T0_mean", "C0_N", "T0_N",  "C0_clus", "T0_clus") {
			*Baseline mean values
			return local disp_stata "``statname''  {c |}"
			local numSpace = `colw' - `len' - 2
		}
		else if inlist("`statname'", "C0_err", "T0_err") {
			*Baseline mean values errors
			return local disp_stata "``statname'' {c |}"
			local numSpace = `colw' - `len' - 1
		}
		else if inlist("`statname'", "1DT", "1DC", "2D") {
			*First difference coefficent
			return local disp_stata "``statname''{c |}"
			local numSpace = `colw' - `len'
		}
		else if inlist("`statname'", "1DT_N", "1DC_N", "2D_N", "1DT_clus", "1DC_clus", "2D_clus") {
			*First difference coefficent
			return local disp_stata "``statname''    {c |}"
			local numSpace = `colw' - `len' - 4
		}
		else if inlist("`statname'", "1DT_err", "1DC_err", "2D_err")  {
			*First difference coefficient error
			return local disp_stata "``statname''   {c |}"
			local numSpace = `colw' - `len' - 3
		}


		**If numbers are so big that numSapce is negative, set it to 0 as
		* numSpace cannot be negative. Table will show incorrectly but
		* will display.
		if `numSpace' < 0 local numSpace 0

		return local disp_pre_space = `numSpace'
		return local disp_len = `len'

	end

	/*************

		Output table in LaTeX

	*************/

	***************
	* Output header
	***************
cap program drop 	outputtex
	program define	outputtex

	syntax varlist, ddtab_resultMap(name) savetex(string) note(string) ///
					[replace onerow starlevels(string) format(string) rwlbls(string) errortype(string) ///
					texdocument texcaption(string) texlabel(string) texnotewidth(numlist) texvspace(string) cluster nonumbers]

		* Replace tex file?
		if "`replace'" != ""		local texreplace	replace

		* Test tex name
		texnametest	, savetex(`savetex')
		local 		  savetex = r(savetex)

		* Create temporary file
		tempname 	texname
		tempfile	texfile

	cap file close `texname'
		file open  `texname' using 	"`texfile'", text write replace
		file write `texname' 		`"%%% Table created in Stata by ieddtable (https://github.com/worldbank/ietoolkit)"' _n _n
		file close `texname'

		* Create preamble for standalone document if option texdocument was selected
		texpreamble	, texname("`texname'") texfile("`texfile'") texcaption("`texcaption'") texlabel("`texlabel'") `texdocument'

		* Write table header
		texheader	, texname("`texname'") texfile("`texfile'") `onerow'  errortype(`errortype') `cluster' `numbers'

		* Write results
		texresults `varlist', ddtab_resultMap(ddtab_resultMap) ///
							  format("`format'") rwlbls("`rwlbls'") errortype("`errortype'") ///
							  texname("`texname'") texfile("`texfile'") ///
							  `onerow'  texvspace("`texvspace'") `cluster'

		* Write row with number of observations if option onerow was selected
		texonerow, ddtab_resultMap(ddtab_resultMap) texname("`texname'") texfile("`texfile'") `onerow' `cluster'

		* Write tex footer
		texfooter, texname("`texname'") texfile("`texfile'") texnotewidth(`texnotewidth') `onerow' errortype(`errortype') note(`note') `texdocument'
		* Save final tex file
		copy "`texfile'" `"`savetex'"', `texreplace'

		* Print confirmation message
		noi di as result `"{phang}Balance table saved to: {browse "`savetex'":`savetex'} "'

end

cap program drop	texresults
	program define	texresults

	syntax	varlist, ddtab_resultMap(name) texname(string) texfile(string) errortype(string) format(string) rwlbls(string) [onerow texvspace(string) cluster]

		****************
		* Prepare inputs
		****************

		*Count numbers of variables to loop over
		local numVars = `:word count `varlist''

		*List of variables to display and loop over when formatting
		local statlist 2D 1DT 1DC C0_mean T0_mean 2D_err 1DT_err 1DC_err C0_err T0_err 2D_N 1DT_N 1DC_N C0_N T0_N

		*Add cluster columns to the list if clusters were used
		if "`cluster'" == "cluster" {
			local statlist 	`statlist' 2D_clus 1DT_clus 1DC_clus C0_clus T0_clus
		}
		* Make sure special characters are displayed correctly in the labels
		local rwlbls = subinstr("`rwlbls'", "&", "\&", .)
		local rwlbls = subinstr("`rwlbls'", "%", "\%", .)
		local rwlbls = subinstr("`rwlbls'", "_", "\_", .)

		* Line spacing
		if "`texvspace'" == ""	{
			local texvspace	3ex
		}
		else {
			* Test if width unit is correctly specified
			local 	vspace_unit = substr("`texvspace'",-2,2)
			if 	!inlist("`vspace_unit'","cm","mm","pt","in","ex","em") {
				noi display as error `"{phang}Option texvspace is incorrectly specified. Vertical space unit must be one of "cm", "mm", "pt", "in", "ex" or "em". For more information, {browse "https://en.wikibooks.org/wiki/LaTeX/Lengths":check LaTeX lengths manual}.{p_end}"'
				error 198
			}

			* Test if width value is correctly specified
			local 	vspace_value = subinstr("`texvspace'","`vspace_unit'","",.)
			capture confirm number `vspace_value'
			if _rc & inlist("`vspace_unit'","cm","mm","pt","in","ex","em") {
				noi display as error "{phang}Option texvspace is incorrectly specified. Vertical space value must be numeric. See {help iebaltab:iebaltab help}. {p_end}"
				error 198
			}
		}


		*********************
		* Loop over variables
		*********************
		forvalues row = 1/`numVars' {

			*Get labels from the list of labels previosly prepared
			local rwlbls = trim(subinstr("`rwlbls'", "@@","", 1))
			gettoken label rwlbls : rwlbls, parse("@@")

			*Prepare the display of each statistic
			foreach stat of local statlist {

				**Run sub command that gets value from matrix and prepares it
				* in the format suitable for the LaTeX table (and adding
				* stars if applicable)
				texdiformat , statname("`stat'") row(`row') ddtab_resultMap(ddtab_resultMap) format("`format'")	errortype(`errortype') `onerow'

				*The main stat
				local `stat' `r(disp_tex)'

			}

			/*****************************************

			 Prepare content of that variable's line

			*****************************************/
			* First column: label
			local 	varline		"`label'"

			* Then other columns with statistics
			foreach column in C0 1DC T0 1DT 2D {

				/***********************************
				 Column with number of observations
				***********************************/

				* Starts with nothing and will only have content if one row was not selected
				local `column'_obs	""

				* If no cluster, add only number of observation
				if "`onerow'" == "" & "`cluster'" == "" {
					local `column'_obs " & ``column'_N'"
				}
				* If cluster, will have two lines: one with number of obs, one with number of clusters
				else if "`onerow'" == "" & "`cluster'" != "" {
					local `column'_obs " & \begin{tabular}[t]{@{}c@{}} ``column'_N'  \\ ``column'_clus'  \end{tabular}"
				}

				/*************************
				 Column with coefficients
				*************************/

				* Get name of stored coef
				if inlist("`column'", "C0", "T0") {
					local coef `column'_mean
				}
				else local coef `column'

				* Starts with nothing, just to make sure
				local `column'_coefs	""

				* If hiding error, add only the coefficient
				if "`errortype'" == "errhide" {
					local `column'_coefs	"& ``coef''"
				}
				* If not, display two lines: one with the coefficient, one the error
				else {
					local `column'_coefs	 "& \begin{tabular}[t]{@{}c@{}} ``coef'' \\ ``column'_err' \end{tabular}"
				}

				/*****************************
				 Add both columns to the line
				******************************/
				local 	varline		`" `varline' ``column'_obs' ``column'_coefs' "'

			}

		* Close variable line with vertical space
		local 	varline		`" `varline' ``column'_obs' ``column'_coefs' \rule{0pt}{`texvspace'}\\"'

		* Write line to tex file
		file open  `texname' using 	"`texfile'", text write append
		file write `texname'		"`varline'" _n
		file close `texname'

	}

end

cap program drop 	texnametest
	program define	texnametest, rclass

	syntax	, savetex(string)

		* Test filename input
		**Find the last . in the file path and assume that
		* the file extension is what follows. If a file path has a . then
		* the file extension must be explicitly specified by the user.

		*Copy the full file path to the file suffix local
		local tex_file_suffix 	= "`savetex'"

		** Find index for where the file type suffix start
		local tex_dot_index 	= strpos("`tex_file_suffix'",".")

		*If no dot then no file extension
		if `tex_dot_index' == 0  local tex_file_suffix 	""

		**If there is one or many . in the file path than loop over
		* the file path until we have found the last one.
		while `tex_dot_index' > 0 {

			*Extract the file index
			local tex_file_suffix 	= substr("`tex_file_suffix'", `tex_dot_index' + 1, .)

			*Find index for where the file type suffix start
			local tex_dot_index 	= strpos("`tex_file_suffix'",".")
		}

		*If no file format suffix is specified, use the default .tex
		if "`tex_file_suffix'" == "" {

			local savetex `"`savetex'.tex"'

		}

		*If a file format suffix is specified make sure that it is one of the two allowed.
		else if !("`tex_file_suffix'" == "tex" | "`tex_file_suffix'" == "txt") {

			noi display as error "{phang}The file format specified in savetex(`name') is other than .tex or .txt. Only those two formats are allowed. If no format is specified .tex is the default. If you have a . in your file path, for example in a folder name, then you must specify the file extension .tex or .txt.{p_end}"
			error 198
		}

		return local savetex `savetex'

end

cap program drop 	texdiformat
	program define	texdiformat, rclass

	syntax , statname(string) row(numlist) ddtab_resultMap(name) format(string) errortype(string) [onerow]

		mat temp = ddtab_resultMap[`row', "`statname'"]
		local `statname' = el(temp,1,1)

		if substr("`statname'", -2,.) == "_N" | substr("`statname'", -5,.) == "_clus" {
			local `statname'	: display %9.0f ``statname''
		}
		else {
			local `statname' 	: display `format' ``statname''
		}

		*Trim spaces on left from left.
		local `statname' = ltrim("``statname''")

		*For coefficients, add stars if applicable
		if inlist("`statname'", "2D", "1DT", "1DC") {

			mat starNumMat = ddtab_resultMap[`row', "`statname'_stars"]
			local starNum = el(starNumMat,1,1)

			if `starNum' == 0 local `statname' "``statname''    "
			if `starNum' == 1 local `statname' "``statname''*   "
			if `starNum' == 2 local `statname' "``statname''**  "
			if `starNum' == 3 local `statname' "``statname''*** "
		}

		*Add parenthesis to errors
		if inlist("`statname'", "2D_err", "1DT_err", "1DC_err", "C0_err", "T0_err") {
			local `statname' "(``statname'')"
		}
		*Add brackets to Number of cluster
		if substr("`statname'", -5,.) == "_clus" {
			local `statname' "{[}``statname'']"
		}

		return local disp_tex "``statname''"
end

cap program drop	texpreamble
	program define	texpreamble

	syntax	, texname(string) texfile(string) [texcaption(string) texlabel(string) texdocument]

		if "`texdocument'"	!= "" {

			file open  `texname' using 	`"`texfile'"', text write append
			file write `texname' 		`"\documentclass{article}"' _n ///
										`""' _n ///
										`"% ----- Preamble "' _n ///
										`"\usepackage[utf8]{inputenc}"' _n ///
										`"\usepackage{adjustbox}"' _n ///
										`"% ----- End of preamble "' _n ///
										`""' _n ///
										`" \begin{document}"' _n ///
										`""' _n ///
										`"\begin{table}[!htbp]"' _n ///
										`"\centering"' _n

			* Write tex caption if specified
			if "`texcaption'" != "" {

				* Make sure special characters are displayed correctly
				local texcaption : subinstr local texcaption "%"  "\%" , all
				local texcaption : subinstr local texcaption "_"  "\_" , all
				local texcaption : subinstr local texcaption "&"  "\&" , all

				file write `texname' 	`"\caption{`texcaption'}"' _n

			}

			* Write tex label if specified
			if "`texlabel'" != "" {

				file write `texname' 	`"\label{`texlabel'}"' _n

			}


			file write `texname'		`"\begin{adjustbox}{max width=\textwidth}"' _n
			file close `texname'
	}

end

cap program drop	texheader
	program define	texheader

	syntax	, texname(string) texfile(string)  errortype(string) [onerow cluster nonumbers]

		** Calculate number of rows

		if ("`onerow'" != "" | "`errortype'" == "errhide") {

			local	toprowcols		2
			local	bottomrowcols	1
			local 	colstring		lccccc
			local	ncol			""
		}
		else {

			local	toprowcols		4
			local	bottomrowcols	2
			local 	colstring		lcccccccccc
			local	ncol			"& N "

			if "`cluster'" != "" {
				local ncol			"`ncol'/{[}Clusters]"
			}
		}

		*Unless option nonumbers is used, number all columns in tex output
		local colnorow = ""
		if "`numbers'" != "nonumbers" {
			local colnomax = strlen("`colstring'") - 1
			forvalues colno = 1/`colnomax' {
				local colnorow = `"`colnorow' & (`colno')"'
			}
			local colnorow = `"`colnorow' \\"'
		}

		*Write part of header that explains which type of errors are displayed
		if "`errortype'" == "errhide" {
			if "`onerow'" != "" {
				local errortitle	""
			}
			else {
				local errortitle	"/N"
			}
		}
		else if "`errortype'" == "se" {
			local errortitle	"/(SE)"
		}
		else if "`errortype'" == "sd" {
			local errortitle	"/(SD)"
		}


		* Write tex header
		file open  `texname' using 	"`texfile'", text write append
		file write `texname'		"\begin{tabular}{@{\extracolsep{5pt}}`colstring'}" _n ///
									"\hline \hline \\[-1.8ex]" _n ///
									"& \multicolumn{`toprowcols'}{c}{Control} & \multicolumn{`toprowcols'}{c}{Treatment}  & \multicolumn{`bottomrowcols'}{c}{Difference-in-differences} \\" _n ///
									"& \multicolumn{`bottomrowcols'}{c}{Baseline} & \multicolumn{`bottomrowcols'}{c}{Difference} & \multicolumn{`bottomrowcols'}{c}{Baseline} & \multicolumn{`bottomrowcols'}{c}{Difference} & \multicolumn{`bottomrowcols'}{c}{} \\" _n ///
									"Variable `ncol' & Mean`errortitle' `ncol' & Coef`errortitle' `ncol' & Mean`errortitle' `ncol' & Coef`errortitle' `ncol' & Coef`errortitle' \\" _n ///
									"`colnorow' \hline" _n
		file close `texname'

end

cap program drop	texfooter
	program define	texfooter

	syntax	, texname(string) texfile(string) errortype(string) note(string) [texnotewidth(numlist) onerow texdocument]

		******************
		* Calculate inputs
		******************

		* Number of columns in the table
		if ("`onerow'" != "" | "`errortype'" == "errhide") {
			local	countcols		6
		}
		else {
			local 	countcols		11
		}

		* Note width
		if "`note'" != "nonote" {
			if "`texnotewidth'" == "" {
				local 	texnotewidth 	1
			}
			else if `texnotewidth' <= 0 {
				noi display as error `"{phang}The value specified in texnotewidth(`texnotewidth') is non-positive. Only positive numbers are allowed. For more information, {net "from http://en.wikibooks.org/wiki/LaTeX/Lengths.smcl":check LaTeX lengths manual}.{p_end}"'
				error 198
			}

			local note = subinstr("`note'", "&", "\&", .)
			local note = subinstr("`note'", "%", "\%", .)
			local note = subinstr("`note'", "_", "\_", .)
		}


		* Note contents
		file open  `texname' using 	"`texfile'", text write append
		file write `texname'		"\hline \hline \\[-1.8ex]" _n

		if "`note'" != "nonote" {
			file write `texname'	"%%% This is the note. If it does not have the correct margins, use texnotewidth() option or change the number before '\textwidth' in line below to fit it to table size." _n ///
									"\multicolumn{`countcols'}{@{} p{`texnotewidth'\textwidth}}" _n ///
									"{\textit{Notes}: `note'}" _n
		}

		file write `texname'		"\end{tabular}" _n

		if "`texdocument'" != "" {
			file write `texname'	"\end{adjustbox}" _n ///
									"\end{table}" _n _n ///
									"\end{document}" _n
		}

		file close `texname'

end

cap program drop	texonerow
	program define	texonerow

	syntax	, texname(string) texfile(string) ddtab_resultMap(name) [onerow cluster]

		if "`onerow'" != "" {

			*Check that all rows have the same number of obs

			*List of variabls to add
			local statlist 2D_N 1DT_N 1DC_N C0_N T0_N

			if "`cluster'" != "" {
				local statlist `statlist'  2D_clus 1DT_clus 1DC_clus C0_clus T0_clus
			}

			*Get their values
			foreach stat of local statlist {

				mat temp = ddtab_resultMap[1, "`stat'"]
				local `stat' = el(temp,1,1)

				local `stat'	: display %9.0f ``stat''

				*Trim spaces on left from left.
				local `stat' = ltrim("``stat''")

			}

				file open  `texname' using 	"`texfile'", text write append
				file write `texname'		"N & `C0_N' & `1DC_N' & `T0_N' & `1DT_N' & `2D_N' \\" _n

			if "`cluster'" != "" {
				file write `texname'		"Clusters & `C0_clus' & `1DC_clus' & `T0_clus' & `1DT_clus' & `2D_clus' \\" _n
			}

				file close `texname'
		}

end
